FormatTranspose ================= 将输入数据从一种存储格式转换为另一种存储格式,如 NCHW ↔ NHWC、NC4HW4、NC8HW8 等,适用于图像或特征图数据。 输入: - **src_data** - 输入数据地址。 - **src_format** - 输入数据格式标识。 - **dst_format** - 输出数据格式标识。 - **batch** - 批大小。 - **channel** - 通道数。 - **plane** - 高*宽。 - **core_mask** - 核掩码(仅适用于共享存储版本)。 输出: - **dst_data** - 格式转换结果地址。 支持平台: ``FT78NE`` ``MT7004`` .. note:: - FT78NE 支持fp, dp, int8, int16, int32, clx64, cplx128 - MT7004 支持hp, fp, i16, i32, cplx64 **共享存储版本:** .. c:function:: void fp_formattranspose_s(int src_format, int dst_format, float* src_data, float* dst_data, int batch, int channel, int plane, int core_mask) .. c:function:: void hp_formattranspose_s(int src_format, int dst_format, half* src_data, half* dst_data, int batch, int channel, int plane, int core_mask) .. c:function:: void dp_formattranspose_s(int src_format, int dst_format, double* src_data, double* dst_data, int batch, int channel, int plane, int core_mask) .. c:function:: void i8_formattranspose_s(int src_format, int dst_format, int8_t* src_data, int8_t* dst_data, int batch, int channel, int plane, int core_mask) .. c:function:: void i16_formattranspose_s(int src_format, int dst_format, int16_t* src_data, int16_t* dst_data, int batch, int channel, int plane, int core_mask) .. c:function:: void i32_formattranspose_s(int src_format, int dst_format, int* src_data, int* dst_data, int batch, int channel, int plane, int core_mask) .. c:function:: void c64_formattranspose_s(int src_format, int dst_format, float* src_data, float* dst_data, int batch, int channel, int plane, int core_mask) .. c:function:: void c128_formattranspose_s(int src_format, int dst_format, double* src_data, double* dst_data, int batch, int channel, int plane, int core_mask) **C调用示例:** .. code-block:: c :linenos: :emphasize-lines: 12 #include #include int main() { float *input = (float *)0xA0000000; // 输入在DDR空间 float *output = (float *)0xC0000000; int batch = 1, channel = 3, plane = 224*224; int src_format = 1; // NHWC int dst_format = 0; // NCHW int core_mask = 0xff; fp_formattranspose_s(src_format, dst_format, input, output, batch, channel, plane, core_mask); return 0; } **私有存储版本:** .. c:function:: void fp_formattranspose_p(int src_format, int dst_format, float* src_data, float* dst_data, int batch, int channel, int plane) .. c:function:: void hp_formattranspose_p(int src_format, int dst_format, half* src_data, half* dst_data, int batch, int channel, int plane) .. c:function:: void dp_formattranspose_p(int src_format, int dst_format, double* src_data, double* dst_data, int batch, int channel, int plane) .. c:function:: void i8_formattranspose_p(int src_format, int dst_format, int8_t* src_data, int8_t* dst_data, int batch, int channel, int plane) .. c:function:: void i16_formattranspose_p(int src_format, int dst_format, int16_t* src_data, int16_t* dst_data, int batch, int channel, int plane) .. c:function:: void i32_formattranspose_p(int src_format, int dst_format, int* src_data, int* dst_data, int batch, int channel, int plane) .. c:function:: void c64_formattranspose_p(int src_format, int dst_format, float* src_data, float* dst_data, int batch, int channel, int plane) .. c:function:: void c128_formattranspose_p(int src_format, int dst_format, double* src_data, double* dst_data, int batch, int channel, int plane) **C调用示例:** .. code-block:: c :linenos: :emphasize-lines: 11 #include #include int main() { float *input = (float *)0x10810000; // 输入在L2空间 float *output = (float *)0x10820000; int batch = 1, channel = 3, plane = 224*224; int src_format = 1; // NHWC int dst_format = 0; // NCHW fp_formattranspose_p(src_format, dst_format, input, output, batch, channel, plane); return 0; }